package leetcode.interviewClassics;

import leetcode.TreeNode;

/**
 * @author Cheng Jun
 * Description: 给你二叉树的根节点 root 和一个表示目标和的整数 targetSum ，判断该树中是否存在 根节点到叶子节点 的路径，
 * 这条路径上所有节点值相加等于目标和 targetSum 。
 * <p>
 * 叶子节点 是指没有子节点的节点。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/path-sum
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * @version 1.0
 * @date 2021/12/2 14:01
 */
public class pathSum1 {
    static boolean flag = false;

    public static void main(String[] args) {
        TreeNode a = new TreeNode(3);
        TreeNode b = new TreeNode(2);
        TreeNode c = new TreeNode(1, a, b);
        System.out.println(hasPathSum(c, 4));
        System.out.println(hasPathSum1(c, 4));
    }

    static boolean hasPathSum(TreeNode root, int targetSum) {
        if (root == null) {
            return false;
        }
        dfs(root, targetSum, 0);
        return flag;
    }

    // sum 表示从根节点到该节点之和，找到 node 是叶子节点，并且sum == targetSum ，返回true
    static void dfs(TreeNode node, int targetSum, int nodeSum) {
        nodeSum += node.val;
        if (node.left == null && node.right == null) {
            if (nodeSum == targetSum) {
                flag = true;
            }
        } else {
            if (node.left != null && !flag) {
                dfs(node.left, targetSum, nodeSum);
            }
            if (node.right != null && !flag) {
                dfs(node.right, targetSum, nodeSum);
            }
        }
    }

    // 叶子节点 没有子节点的节点。
    // 用减法思想，如果存在满足条件的路径，那么 targetSum 不断减去当前节点值，到最后 肯定和叶子节点值相等
    static boolean hasPathSum1(TreeNode root, int targetSum) {
        if (root == null) {
            return false;
        }
        // 到达叶子节点，判断当前目标值和 节点值是否相等
        if (root.left == null && root.right == null) {
            return root.val == targetSum;
        }
        return hasPathSum1(root.left, targetSum - root.val) || hasPathSum1(root.right, targetSum - root.val);
    }


}
